package cn.yu.takeoutbackend.fileManagement.service.impl;

import cn.hutool.core.date.DateUtil;
import cn.hutool.core.lang.UUID;
import cn.hutool.core.util.IdUtil;
import cn.yu.takeoutbackend.fileManagement.dao.FileManagementDao;
import cn.yu.takeoutbackend.fileManagement.entity.File;
import cn.yu.takeoutbackend.fileManagement.service.FileManagementService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;

import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.StandardCopyOption;

@Service
@Slf4j
public class FileManagementServiceImpl extends ServiceImpl<FileManagementDao, File> implements FileManagementService {

    @Autowired
    @Qualifier("fileStorageLocation")
    private Path fileStorageLocation;

    // 定义允许的最大文件大小（例如：10MB）
    private static final long MAX_FILE_SIZE = 30 * 1024 * 1024;

    @Override
    public String updateFile(MultipartFile file) {
        try {
            // 校验文件是否为空
            if (file == null || file.isEmpty()) {
                throw new IllegalArgumentException("上传的文件不能为空");
            }

            // 获取文件原始名称
            final String originalName = file.getOriginalFilename();
            if (originalName == null || originalName.isEmpty()) {
                throw new IllegalArgumentException("文件名不能为空");
            }

            // 校验文件名合法性
            validateFileName(originalName);

            // 校验文件大小
            if (file.getSize() > MAX_FILE_SIZE) {
                throw new IllegalArgumentException("文件大小超出限制，最大支持 " + MAX_FILE_SIZE / (1024 * 1024) + "MB");
            }

            // 组装文件信息
            String fileExtension = getFileExtension(originalName);
            String storeName = generateUniqueFileName(fileExtension);

            Path datePath = this.fileStorageLocation.resolve(String.valueOf(DateUtil.today()));
            Files.createDirectories(datePath);

            final Path targetLocation = datePath.resolve(storeName);

            // 存储文件
            Files.copy(file.getInputStream(), targetLocation, StandardCopyOption.REPLACE_EXISTING);

            // 入库
            final File storeFile = new File();
            storeFile.setOriginalName(originalName);
            storeFile.setId(IdUtil.getSnowflakeNextIdStr());
            storeFile.setFileType(fileExtension);
            storeFile.setFileSize(file.getSize());
            storeFile.setStoredName(storeName);
            storeFile.setFilePath(targetLocation.toAbsolutePath().toString()); // 存储相对路径
            final boolean save = this.save(storeFile);

            if (!save) {
                log.error("文件上传失败，文件名: {}, 文件大小: {}", originalName, file.getSize());
                return null;
            }

            return storeFile.getId();

        } catch (IllegalArgumentException e) {
            log.warn("文件上传参数校验失败: {}", e.getMessage());
            throw e; // 将参数校验异常抛出，便于调用方处理
        } catch (IOException e) {
            log.error("文件上传失败，IO异常: {}", e.getMessage(), e);
            return null;
        } catch (Exception e) {
            log.error("文件上传失败，未知异常: {}", e.getMessage(), e);
            return null;
        }
    }

    /**
     * 校验文件名合法性
     *
     * @param fileName 文件名
     */
    private void validateFileName(String fileName) {
        if (fileName.contains("..") || fileName.contains("/") || fileName.contains("\\") || fileName.contains(":")) {
            throw new IllegalArgumentException("文件名包含非法字符: " + fileName);
        }
    }

    /**
     * 获取文件扩展名
     *
     * @param fileName 文件名
     * @return 文件扩展名
     */
    private String getFileExtension(String fileName) {
        int dotIndex = fileName.lastIndexOf(".");
        return (dotIndex == -1) ? "" : fileName.substring(dotIndex);
    }

    /**
     * 生成唯一文件名
     *
     * @param fileExtension 文件扩展名
     * @return 唯一文件名
     */
    private String generateUniqueFileName(String fileExtension) {
        return UUID.randomUUID().toString().replace("-", "") + System.currentTimeMillis() + fileExtension;
    }

    /**
     * 获取文件的相对路径
     *
     * @param absolutePath 文件的绝对路径
     * @return 相对路径
     */
    private String getRelativePath(Path absolutePath) {
        return fileStorageLocation.relativize(absolutePath).toString();
    }
}
